package com.yeung.swordoffer;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;

/**
 * 小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。
 * 但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。
 * 没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。
 * 现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
 */
public class 和为S的连续正数序列 {
    public static void main(String[] args) {
        FindContinuousSequence(3);
    }

    //输出所有和为S的连续正数序列。序列内按照从小至大的顺序，序列间按照开始数字从小到大的顺序
    public static ArrayList<ArrayList<Integer>> FindContinuousSequence(int sum) {
        ArrayList<ArrayList<Integer>> all = new ArrayList<>();
        for (int i = 1; i <= (sum / 2) + 1; i++) {
            ArrayList<Integer> tmp = new ArrayList<>();
            getResult(i, 0, sum, all, tmp);
        }
        all.removeIf(list -> list.size() == 0);
        Collections.sort(all, new Comparator<ArrayList<Integer>>() {
            @Override
            public int compare(ArrayList<Integer> o1, ArrayList<Integer> o2) {
                Integer l1 = o1.get(0);
                Integer l2 = o2.get(0);
                return l1 - l2;
            }
        });
        return all;
    }

    public static void getResult(int n, int sum, int target, ArrayList<ArrayList<Integer>> all, ArrayList<Integer> tmp) {
        if (n > target) return;
        if (sum + n == target) {
            tmp.add(n);
            if (tmp.size() > 1)
                all.add(tmp);
            return;
        }
        if (sum + n < target) {
            tmp.add(n);
            getResult(n + 1, sum + n, target, all, tmp);
        }
    }
}
